#!/usr/bin/env python3
"""简单测试嵌入模型"""
from pathlib import Path
import sys
project_root = Path(__file__).parent.parent
sys.path.append(str(project_root))
from core.embedding_manager import embedding_manager

def simple_test():
    print("=== 简单测试 ===")
    
    # 检查嵌入管理器状态
    print(f"嵌入管理器就绪: {embedding_manager.is_ready()}")
    
    if embedding_manager.is_ready():
        try:
            # 测试嵌入功能
            print("测试嵌入功能...")
            result = embedding_manager.test_embedding("测试文本")
            print(f"嵌入测试结果: {result}")
            
            if result:
                # 尝试嵌入一个简单的文档
                print("尝试嵌入文档...")
                docs = ["这是第一个测试文档", "这是第二个测试文档"]
                embeddings = embedding_manager.embed_documents(docs)
                print(f"嵌入结果: {len(embeddings)} 个向量")
                if embeddings:
                    print(f"第一个向量维度: {len(embeddings[0])}")
            
        except Exception as e:
            print(f"测试过程中出错: {e}")
            import traceback
            traceback.print_exc()

if __name__ == "__main__":
    simple_test() 